﻿import requests
import json

print(' 获取Chroma数据库中的所有文档...')
print('=' * 80)

try:
    response = requests.get('http://localhost:5000/api/get_all_documents')
    if response.status_code == 200:
        result = response.json()
        print(f' 成功获取 {result["total_count"]} 个文档')
        print(f' 状态: {result["status"]}')
        print(f' 消息: {result["message"]}')
        print('=' * 80)
        
        for i, doc in enumerate(result['documents'], 1):
            print(f'\n 文档 {i}:')
            print(f'   ID: {doc["id"]}')
            print(f'   标题: {doc["metadata"].get("title", "无标题")}')
            print(f'   标准: {doc["metadata"].get("standard_code", "无标准")}')
            print(f'    分类: {doc["metadata"].get("category", "无分类")}')
            print(f'   发布日期: {doc["metadata"].get("publish_date", "无日期")}')
            print(f'   内容: {doc["document"]}')
            print('-' * 80)
    else:
        print(f' 获取失败: {response.status_code}')
except Exception as e:
    print(f' 请求失败: {e}')

print('\n 数据库统计信息:')
try:
    response = requests.get('http://localhost:5000/api/health')
    if response.status_code == 200:
        result = response.json()
        print(f'   文档总数: {result["collection_count"]}')
        print(f'   Chroma版本: {result["chroma_version"]}')
        print(f'   服务状态: {result["message"]}')
except Exception as e:
    print(f' 获取统计信息失败: {e}')
